
gpu_id=1

log_date=$(date '+%Y-%m-%d_%H:%M:%S')
tensorboard_logdir=saved/SciRE/event_${log_date}


saved_data=saved/SciRE/SciRE.pkl
CUDA_VISIBLE_DEVICES=${gpu_id} python train.py \
        --saved_model='saved/SciRE/SciRE_model.pkl' \
        --saved_data=${saved_data} \
        --tensorboard_logdir=${tensorboard_logdir} \
        --dataset='SciRE' \
        --train_file='train.txt' \
        --dev_file='test.txt'\
        --test_file='test.txt' \
        --relation2id='rel2id.txt' \
        --gaz_emb='./lexicon/ctb.50d.vec'\
        --bichar_emb='./lexicon/gigaword_chn.all.a2b.bi.ite50.vec'\
        --char_emb='./lexicon/gigaword_chn.all.a2b.uni.ite50.vec'\
        --epochs=50 \
        --batch_size=64 \
        --optimizer='AdamW'\
        --warmup_type='linear' \
        --learning_rate=5e-5\
        --fuse_type='concat'\


# saved_data=saved/SciRE/SciRE_withNA.pkl
# CUDA_VISIBLE_DEVICES=${gpu_id} python train.py \
#         --saved_model='saved/SciRE/SciRE_model.pkl' \
#         --saved_data=${saved_data} \
#         --tensorboard_logdir=${tensorboard_logdir} \
#         --dataset='SciRE_withNA' \
#         --train_file='train.txt' \
#         --dev_file='test.txt'\
#         --test_file='test.txt' \
#         --relation2id='rel2id.txt' \
#         --gaz_emb='./lexicon/ctb.50d.vec'\
#         --bichar_emb='./lexicon/gigaword_chn.all.a2b.bi.ite50.vec'\
#         --char_emb='./lexicon/gigaword_chn.all.a2b.uni.ite50.vec'\
#         --epochs=50 \
#         --batch_size=64 \
#         --optimizer='AdamW'\
#         --warmup_type='linear' \
#         --learning_rate=5e-5\
#         --fuse_type='concat'\